/*
 * Copyright (C) 2011-2013 GUIGUI Simon, fyhertz@gmail.com
 *
 * This file is part of Spydroid (http://code.google.com/p/spydroid-ipcamera/)
 *
 * Spydroid is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This source code is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this source code; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

package net.majorkernelpanic.streaming.video;

import java.util.ArrayList;
import java.util.HashMap;

import android.annotation.SuppressLint;
import android.media.MediaCodecInfo;
import android.media.MediaCodecList;
import android.os.Build;
import android.util.Log;
import android.util.SparseArray;

@SuppressLint("InlinedApi")
public class CodecManager {

    public final static String TAG = "CodecManager";

    public static final int[] SUPPORTED_COLOR_FORMATS = {
            MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420Planar,
            MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420SemiPlanar
    };

    /**
     * There currently is no way to know if an encoder is software or hardware from the MediaCodecInfo class,
     * so we need to maintain a list of known software encoders.
     */
    public static final String[] SOFTWARE_ENCODERS = {
            "OMX.google.h264.encoder"
    };

    /**
     * Contains a list of encoders and color formats that we may use with a {@link CodecManager.Translator}.
     */
    static class Codecs {
        /**
         * A hardware encoder supporting a color format we can use.
         */
        public String hardwareCodec;
        public int hardwareColorFormat;
        /**
         * A software encoder supporting a color format we can use.
         */
        public String softwareCodec;
        public int softwareColorFormat;
    }

    /**
     * Contains helper functions to choose an encoder and a color format.
     */
    static class Selector {

        private static HashMap<String, SparseArray<ArrayList<String>>> sHardwareCodecs = new HashMap<>();
        private static HashMap<String, SparseArray<ArrayList<String>>> sSoftwareCodecs = new HashMap<>();

        /**
         * Determines the most appropriate encoder to compress the video from the Camera
         */
        public static Codecs findCodecsFormMimeType(String mimeType, boolean tryColorFormatSurface) {
            findSupportedColorFormats(mimeType);
            SparseArray<ArrayList<String>> hardwareCodecs = sHardwareCodecs.get(mimeType);
            SparseArray<ArrayList<String>> softwareCodecs = sSoftwareCodecs.get(mimeType);
            Codecs list = new Codecs();

            // On devices running 4.3, we need an encoder supporting the color format used to work with a Surface
            if (Build.VERSION.SDK_INT >= 18 && tryColorFormatSurface) {
                int colorFormatSurface = MediaCodecInfo.CodecCapabilities.COLOR_FormatSurface;
                try {
                    // We want a hardware encoder
                    list.hardwareCodec = hardwareCodecs.get(colorFormatSurface).get(0);
                    list.hardwareColorFormat = colorFormatSurface;
                } catch (Exception e) {
                }
                try {
                    // We want a software encoder
                    list.softwareCodec = softwareCodecs.get(colorFormatSurface).get(0);
                    list.softwareColorFormat = colorFormatSurface;
                } catch (Exception e) {
                }

                if (list.hardwareCodec != null) {
                    Log.v(TAG, "Choosen primary codec: " + list.hardwareCodec + " with color format: " + list.hardwareColorFormat);
                } else {
                    Log.e(TAG, "No supported hardware codec found !");
                }
                if (list.softwareCodec != null) {
                    Log.v(TAG, "Choosen secondary codec: " + list.hardwareCodec + " with color format: " + list.hardwareColorFormat);
                } else {
                    Log.e(TAG, "No supported software codec found !");
                }
                return list;
            }

            for (int i = 0; i < SUPPORTED_COLOR_FORMATS.length; i++) {
                try {
                    list.hardwareCodec = hardwareCodecs.get(SUPPORTED_COLOR_FORMATS[i]).get(0);
                    list.hardwareColorFormat = SUPPORTED_COLOR_FORMATS[i];
                    break;
                } catch (Exception e) {
                }
            }
            for (int i = 0; i < SUPPORTED_COLOR_FORMATS.length; i++) {
                try {
                    list.softwareCodec = softwareCodecs.get(SUPPORTED_COLOR_FORMATS[i]).get(0);
                    list.softwareColorFormat = SUPPORTED_COLOR_FORMATS[i];
                    break;
                } catch (Exception e) {
                }
            }

            if (list.hardwareCodec != null) {
                Log.v(TAG, "Choosen primary codec: " + list.hardwareCodec + " with color format: " + list.hardwareColorFormat);
            } else {
                Log.e(TAG, "No supported hardware codec found !");
            }
            if (list.softwareCodec != null) {
                Log.v(TAG, "Choosen secondary codec: " + list.hardwareCodec + " with color format: " + list.softwareColorFormat);
            } else {
                Log.e(TAG, "No supported software codec found !");
            }

            return list;
        }

        /**
         * Returns an associative array of the supported color formats and the names of the encoders for a given mime type
         * This can take up to sec on certain phones the first time you run it...
         **/
        @SuppressLint("NewApi")
        static private void findSupportedColorFormats(String mimeType) {
            SparseArray<ArrayList<String>> softwareCodecs = new SparseArray<ArrayList<String>>();
            SparseArray<ArrayList<String>> hardwareCodecs = new SparseArray<ArrayList<String>>();

            if (sSoftwareCodecs.containsKey(mimeType)) {
                return;
            }

            Log.v(TAG, "Searching supported color formats for mime type \"" + mimeType + "\"...");

            // We loop through the encoders, apparently this can take up to a sec (testes on a GS3)
            for (int j = MediaCodecList.getCodecCount() - 1; j >= 0; j--) {
                MediaCodecInfo codecInfo = MediaCodecList.getCodecInfoAt(j);
                if (!codecInfo.isEncoder()) continue;

                String[] types = codecInfo.getSupportedTypes();
                for (int i = 0; i < types.length; i++) {
                    if (types[i].equalsIgnoreCase(mimeType)) {
                        MediaCodecInfo.CodecCapabilities capabilities = codecInfo.getCapabilitiesForType(mimeType);

                        boolean software = false;
                        for (int k = 0; k < SOFTWARE_ENCODERS.length; k++) {
                            if (codecInfo.getName().equalsIgnoreCase(SOFTWARE_ENCODERS[i])) {
                                software = true;
                            }
                        }

                        // And through the color formats supported
                        for (int k = 0; k < capabilities.colorFormats.length; k++) {
                            int format = capabilities.colorFormats[k];
                            if (software) {
                                if (softwareCodecs.get(format) == null)
                                    softwareCodecs.put(format, new ArrayList<String>());
                                softwareCodecs.get(format).add(codecInfo.getName());
                            } else {
                                if (hardwareCodecs.get(format) == null)
                                    hardwareCodecs.put(format, new ArrayList<String>());
                                hardwareCodecs.get(format).add(codecInfo.getName());
                            }
                        }

                    }
                }
            }

            // Logs the supported color formats on the phone
            StringBuilder e = new StringBuilder();
            e.append("Supported color formats on this phone: ");
            for (int i = 0; i < softwareCodecs.size(); i++)
                e.append(softwareCodecs.keyAt(i) + ", ");
            for (int i = 0; i < hardwareCodecs.size(); i++)
                e.append(hardwareCodecs.keyAt(i) + (i == hardwareCodecs.size() - 1 ? "." : ", "));
            Log.v(TAG, e.toString());

            sSoftwareCodecs.put(mimeType, softwareCodecs);
            sHardwareCodecs.put(mimeType, hardwareCodecs);
            return;
        }


    }

    static class Translator {

        private int mOutputColorFormat;
        private int mWidth;
        private int mHeight;
        private int mYStride;
        private int mUVStride;
        private int mYSize;
        private int mUVSize;
        private int bufferSize;
        private int i;
        private byte[] tmp;

        public Translator(int outputColorFormat, int width, int height) {
            mOutputColorFormat = outputColorFormat;
            mWidth = width;
            mHeight = height;
            mYStride = (int) Math.ceil(mWidth / 16.0) * 16;
            mUVStride = (int) Math.ceil((mYStride / 2) / 16.0) * 16;
            mYSize = mYStride * mHeight;
            mUVSize = mUVStride * mHeight / 2;
            bufferSize = mYSize + mUVSize * 2;
            tmp = new byte[mUVSize * 2];
        }

        public int getBufferSize() {
            return bufferSize;
        }

        public int getUVStride() {
            return mUVStride;
        }

        public int getYStride() {
            return mYStride;
        }

        public byte[] translate(byte[] buffer) {

            if (mOutputColorFormat == MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420Planar) {
                // FIXME: May be issues because of padding here :/
                int wh4 = bufferSize / 6; //wh4 = width*height/4
                byte tmp;
                for (i = wh4 * 4; i < wh4 * 5; i++) {
                    tmp = buffer[i];
                    buffer[i] = buffer[i + wh4];
                    buffer[i + wh4] = tmp;
                }
            } else if (mOutputColorFormat == MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420SemiPlanar) {
                // We need to interleave the U and V channel
                System.arraycopy(buffer, mYSize, tmp, 0, mUVSize * 2); // Y
                for (i = 0; i < mUVSize; i++) {
                    buffer[mYSize + i * 2] = tmp[i + mUVSize]; // Cb (U)
                    buffer[mYSize + i * 2 + 1] = tmp[i]; // Cr (V)
                }
            }

            return buffer;
        }


    }


}
